package com.codejiwei.core.rdd

import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}

object Spark_Operator_Glom {
  def main(args: Array[String]): Unit = {
    val conf = new SparkConf().setMaster("local[1]").setAppName("Glom")
    val sc = new SparkContext(conf)
    val rdd = sc.makeRDD(List(1, 2, 3, 4, 5), 2)

    //TODO glom()  将同一个分区的数据直接转换为相同类型的内存数组进行处理，分区不变
    val rdd1: RDD[Array[Int]] = rdd.glom()


//    rdd1.collect().foreach(println)
    rdd1.collect().foreach(data => println(data.mkString(", ")))

    /*
    TODO 小功能：计算所有分区最大值求和（分区内取最大值，分区间最大值求和）
    */
    val rdd2: RDD[Int] = sc.makeRDD(List(1, 2, 3, 4, 5, 6), 3)
    val result = rdd2.mapPartitions(value => {
      List(value.max).iterator
    })
      .reduce(_ + _)
    println(result)

    val result2 = rdd2.glom()
      .map(_.max)
      .reduce(_ + _)
    println(result2)



    sc.stop()
  }
}
